Python code for CFD